//
// Copyright (C) 2007 United States Government as represented by the
// Administrator of the National Aeronautics and Space Administration
// (NASA).  All Rights Reserved.
// 
// This software is distributed under the NASA Open Source Agreement
// (NOSA), version 1.3.  The NOSA has been approved by the Open Source
// Initiative.  See the file NOSA-1.3-JPF at the top of the distribution
// directory tree for the complete NOSA document.
// 
// THE SUBJECT SOFTWARE IS PROVIDED "AS IS" WITHOUT ANY WARRANTY OF ANY
// KIND, EITHER EXPRESSED, IMPLIED, OR STATUTORY, INCLUDING, BUT NOT
// LIMITED TO, ANY WARRANTY THAT THE SUBJECT SOFTWARE WILL CONFORM TO
// SPECIFICATIONS, ANY IMPLIED WARRANTIES OF MERCHANTABILITY, FITNESS FOR
// A PARTICULAR PURPOSE, OR FREEDOM FROM INFRINGEMENT, ANY WARRANTY THAT
// THE SUBJECT SOFTWARE WILL BE ERROR FREE, OR ANY WARRANTY THAT
// DOCUMENTATION, IF PROVIDED, WILL CONFORM TO THE SUBJECT SOFTWARE.
//
package gov.nasa.jpf.symbc.uberlazy;

import gov.nasa.jpf.symbc.heap.HeapChoiceGenerator;

public class PartitionChoiceGenerator extends HeapChoiceGenerator {

	EquivalenceObjects [] eqObjsList; // list of equivalence classes
	
	public PartitionChoiceGenerator(int size) {
		super(size);
		eqObjsList = new EquivalenceObjects[size];
	}
	
	public void setEquivalenceObj(EquivalenceObjects equivObj) {
		eqObjsList[getNextChoice()] = equivObj;
	}
	
	// creates a deep copy 
	public EquivalenceObjects getCurrentEquivalenceObject() {

		EquivalenceObjects eqObjs;
		
		eqObjs = eqObjsList[getNextChoice()];
		if(eqObjs != null) {
			return ((EquivalenceObjects)eqObjs.make_copy());
		} else {
			return null;
		}
	}
}